python code for accepting user input